package demo4.mapper;

import demo4.pojo.User;
import demo4.util.IDBUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

@Component
public class UserMapper implements iUserMapper {
    @Autowired
    private IDBUtil dbUtil;
    @Override
    public int insertUser(User user) {
        System.out.println("UserMapper:将用户信息插入数据库中....");
        try {
            Connection connection = dbUtil.getConnection();
            String sql = "INSERT INTO user(username,password,nickname,age) "+
                    "VALUES(?,?,?,?)";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNickname());
            ps.setInt(4,user.getAge());
            return ps.executeUpdate();

        } catch (SQLException e) {
            e.printStackTrace();
        }
        return 0;
    }

    @Override
    public User selectUserByUsername(String username) {
        try( Connection connection = dbUtil.getConnection();) {
           String sql = "SELECT id,username,password,nickname,age " +
                   "FROM user " +
                   "WHERE username=?";
           PreparedStatement ps = connection.prepareStatement(sql);
           ps.setString(1,username);
            ResultSet rs = ps.executeQuery();
            if(rs.next()){
                User user = new User();
                user.setId(rs.getInt("id"));
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                return user;
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }

        return null;
    }

    @Override
    public List<User> selectAllUser() {
        List< User>list = new ArrayList<>();
        try(Connection connection = dbUtil.getConnection();) {
            String sql = "SELECT id,username,password,nickname,age " +
                    "FROM user";
            PreparedStatement ps = connection.prepareStatement(sql);
            ResultSet rs = ps.executeQuery();
            while(rs.next()){
                User user = new User();
                user.setId(rs.getInt("id"));
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                list.add(user);


            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return list;
    }

    @Override
    public int updateUser(User user) {
        try(Connection connection = dbUtil.getConnection();) {
            String sql = "UPDATE user " +
                    "SET username=?,password=?,nickname=?,age=? " +
                    "WHERE id=?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNickname());
            ps.setInt(4,user.getAge());
            ps.setInt(5,user.getId());
            return ps.executeUpdate();
        }catch(SQLException e){
            e.printStackTrace();
        }
        return 0;
    }

}





